python pretrain.py
python main.py